from numpy import *
import bayes

postList, postClasses = bayes.loadDataSet()
# print postList
print postClasses

vocabList = bayes.createVocabList(postList)
print vocabList

# vocabVect = bayes.setOfWords2Vec(vocabList, postList[0])
# print vocabVect

# trainMat = []
# for post in postList:
#     trainMat.append(bayes.setOfWords2Vec(vocabList, post))
#
# p0V, p1V, pAb = bayes.trainNB0(trainMat, postClasses)
# print p0V
# print p1V
# print pAb
# print log(pAb)

# bayes.testingNB()
bayes.spamTest()
